from zhipuai import ZhipuAI
import json
import requests
# 这个api是根据公司名称获得该公司所有注册信息
client = ZhipuAI(api_key="8fb21e517a965b10cf87b7fdadf18a74.UVLbxTJXdL3YLxcT") # 请填写您自己的APIKey
# 先获取公司名称
tools = [
    {
        "type": "function",
        "function": {
            "name": "get_company_register",
            "description": "根据公司名称获得该公司所有注册信息",
            "parameters": {
                "type": "object",
                "properties": {
                    "company_name": {
                        "type": "string",
                        "description": "基本字段",
                    }
                },
                "required": ["company_name"],
            },
        }
    }
]

headers = {
    'Content-Type': 'application/json',
    'Authorization': 'Bearer DE32EB46D23350E0B75B3702C8D5DC036F8CE32DC73C919D'
}
# 下面的代码根据获得的key和value和tools里面的api名称来获取公司的名字

domain = "comm.chatglm.cn"
url = f"https://{domain}/law_api/get_company_register"

data = {
    "company_name": "舟山市龙马环卫环境工程有限公司"
}

rsp = requests.post(url, json=data, headers=headers)
print(rsp.json())
